////////////////////////////////////////////////
// DESCRIPTION:
//    Win32 DLL Module Wrapper
//
// Legal Notices:
//     Copyright (c) 2008, Telliam Consulting, LLC.
//     All rights reserved.
//
//     Redistribution and use in source and binary forms, with or without modification,
//     are permitted provided that the following conditions are met:
//
//     * Redistributions of source code must retain the above copyright notice,
//       this list of conditions and the following disclaimer.
//
//     * Redistributions in binary form must reproduce the above copyright notice,
//       this list of conditions and the following disclaimer in the documentation
//       and/or other materials provided with the distribution.
//
//     * Neither the name of Telliam Consulting nor the names of its contributors
//       may be used to endorse or promote products derived from this software
//       without specific prior written permission.
//
// Disclaimer:
//     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
//     ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
//     WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
//     IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
//     INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
//     BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
//     DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
//     LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
//     OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
//     OF THE POSSIBILITY OF SUCH DAMAGE.  
//
#include <windows.h>
#include <FusionWare\Win32.h>

using namespace FusionWare::Modules;

CDllModule::CDllModule()
: hModule(NULL)
{
}

CDllModule::~CDllModule()
{
    Unload();
}


bool CDllModule::Load(LPCTSTR Path, BOOL ZoneFlag, bool Driver)
{
#ifdef UNDER_CE
   if(Driver)
       this->hModule = LoadDriver(Path);
   else
#endif
      this->hModule = LoadLibrary(Path);

   if(NULL == this->hModule)
   {
       ERRORMSG(ZoneFlag, (_T("Failed to load DLL \"%s\"\n"),Path));
       return false;
   }

   return true;
}

bool CDllModule::Unload()
{
    BOOL retVal = TRUE;
    if(NULL!=this->hModule)
    {  
        retVal = FreeLibrary(this->hModule);
        this->hModule = NULL;
    }

    return retVal != FALSE;
}

bool CDllModule::GetProcAddresses(void* pBase, BOOL ZoneFlag)
{
    const ImportFunctionEntry* pEntries = GetFunctionEntries();
    bool retVal = true;

    while(pEntries!=NULL && pEntries->pFuncName!=NULL)
    {
        FARPROC pFunc = GetAddress(pEntries->pFuncName);
        if(pFunc == NULL)
        {
            // If it's required then flag the fail but
            // keep going. This helps identify all missing
            // entries in one pass to simplify debugging.
            if(pEntries->Required)
            {
                ERRORMSG(ZoneFlag, (_T("Function \"%s\" not found\n"),pEntries->pFuncName));
                retVal = false;
                pFunc = NULL;
            }    
            else
            {
                // otherwise get the stub pointer.
                // the Stub is allowed to be NULL for optional
                // entry points that don't require a stub. 
                pFunc = pEntries->Stub;
            }
        }

        // store the FARPROC value at the specified offset from base. 
        *((FARPROC*)((BYTE*)pBase + pEntries->Offset)) = pFunc;
        pEntries++;
    }
    return retVal;
}
